package code1.binary_tree;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * 原题链接：https://leetcode.cn/problems/binary-tree-level-order-traversal/
 */
public class P102 {

    //层序遍历
    public List<List<Integer>> levelOrder(TreeNode root) {
        Queue<TreeNode> queue=new LinkedList<>();
        List<List<Integer>> result=new ArrayList<>();
        if (root==null){
            return result;
        }
        queue.add(root);
        while(!queue.isEmpty()){
            int size= queue.size();             //当前层的元素个数
            List<Integer> list=new ArrayList<>();
            for(int i=0;i<size;i++){
                TreeNode node = queue.poll();
                list.add(node.val);
                if(node.left!=null){
                    queue.offer(node.left);     //往队列中添加左节点
                }
                if(node.right!=null){
                    queue.offer(node.right);    //往队列中添加右节点
                }
            }
            result.add(list);
        }
        return result;
    }
}
